package com.moran.deepseek.domain;

import lombok.Data;
import java.util.List;

@Data
public class ChatRequest {
    private List<Message> messages;
    private String model = "deepseek-chat";
    private Double frequency_penalty = 0.0;
    private Integer max_tokens = 2048;
    private Double presence_penalty = 0.0;
    private ResponseFormat response_format = new ResponseFormat("text");
    private List<String> stop = null;
    private Boolean stream = false;
    private Object stream_options = null;
    private Double temperature = 1.0;
    private Double top_p = 1.0;
    private List<Object> tools = null;
    private String tool_choice = "none";
    private Boolean logprobs = false;
    private Integer top_logprobs = null;

    @Data
    public static class Message {
        private String role;
        private String content;

        public Message(String role, String content) {
            this.role = role;
            this.content = content;
        }
    }

    @Data
    public static class ResponseFormat {
        private String type;

        public ResponseFormat(String type) {
            this.type = type;
        }
    }
}